package com.example.java_gobang.config;

import com.example.java_gobang.api.controller.GameAPI;
import com.example.java_gobang.api.controller.MatchAPI;
import com.example.java_gobang.api.controller.TestAPI;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.socket.config.annotation.EnableWebSocket;
import org.springframework.web.socket.config.annotation.WebSocketConfigurer;
import org.springframework.web.socket.config.annotation.WebSocketHandlerRegistry;
import org.springframework.web.socket.server.support.HttpSessionHandshakeInterceptor;

/**
 * WebSocket 配置类，用于配置 WebSocket 处理器和路径。
 */
@Configuration
@EnableWebSocket
public class WebSocketConfig implements WebSocketConfigurer {

    @Autowired
    private TestAPI testAPI;   // 注入 TestAPI WebSocket 处理器

    @Autowired
    private MatchAPI matchAPI; // 注入 MatchAPI WebSocket 处理器

    @Autowired
    private GameAPI gameAPI;   // 注入 GameAPI WebSocket 处理器

    /**
     * 注册 WebSocket 处理器及其相关配置。
     *
     * @param registry WebSocketHandlerRegistry 用于注册 WebSocket 处理器
     */
    @Override
    public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
        // 注册 TestAPI 处理器，处理 "/test" 路径的 WebSocket 请求
        registry.addHandler(testAPI, "/test");

        // 注册 MatchAPI 处理器，处理 "/findMatch" 路径的 WebSocket 请求
        // 并添加 HttpSessionHandshakeInterceptor 拦截器以支持 HTTP 会话的 WebSocket 握手
        registry.addHandler(matchAPI, "/findMatch")
                .addInterceptors(new HttpSessionHandshakeInterceptor());

        registry.addHandler(gameAPI, "/game")
                .addInterceptors(new HttpSessionHandshakeInterceptor());
    }
}
